package cn.tedu.v5;

import java.io.*;
import java.net.*;
import java.util.*;
/**
 * Version  5   登录与退出   功能：可以加在v3
 *
  */

public class Server {
    private static final Integer PORT=1234;
    private ServerSocket serverSocket;
    //理解：1.服务器端发提示---写出客户端：集合保存所有用户
    private List<PrintWriter> allOut=new ArrayList<>();
    //1.私有化属性：存账号  密码
    private Map<String,String> userDatabase;
    public Server() {
        try {
            serverSocket = new ServerSocket(PORT);
            //2.创建Map对象
            userDatabase=new HashMap<>();
            userDatabase.put("皮皮","99999");
            userDatabase.put("屁屁","88888");
        } catch (BindException e) {
            System.out.println();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
    public void start()  {
        try {
            while (true) {
                System.out.println("等待客户端的连接");
               Socket socket=serverSocket.accept();

                new Thread(new ClientHandler(socket)).start();
            }
        } catch (IOException e) {
            System.out.println("客户端链接失败了");
        } finally {
            if (serverSocket!=null&&serverSocket.isClosed()){
                try {
                    serverSocket.close();
                } catch (IOException e) {
                    System.out.println("服务器关闭失败");
                }
                System.out.println("服务器关闭成功");
            }
        }

    }
    public static void main(String[] args) {
        new Server().start();
    }
    //5.5  因为用户账号可能有很多  封装Login       传入参数判断
    private boolean Login(String user,String pass){
        return userDatabase.containsKey(user)&&userDatabase.get(user).equals(pass);
    }
    //6. handleChat   封装：读写功能   ---放入线程中


    private class ClientHandler implements Runnable {
        private Socket socket;
        private String ip;
        private PrintWriter pw;
        private BufferedReader br;
        //4.线程当中：确定哪个客户端---赋予功能
        private String username;
        private String password;



        public ClientHandler(Socket s){
            this.socket=s;
            ip=socket.getInetAddress().getHostAddress();
            try {
                br = new BufferedReader(new InputStreamReader(socket.getInputStream(), "UTF8"));
                pw=new PrintWriter(new BufferedWriter(new OutputStreamWriter(socket.getOutputStream(),"UTF8")),true);
                synchronized (allOut){//监视集合中 只允许一个客户端过来
                    //来客户端 添加写出的方法
                    allOut.add(pw);//目的：客户端  写出消息---分发消息
                }
            } catch (IOException e) {
                System.out.println("客户端断开连接了");
            }
        }
        //同步：broadcastMessage消息广播功能
        //参数：String message  转发给客户端   文本
        //增强for：目的：来一个 遍历一个客户端  给这个客户端加消息
        private synchronized void broadcastMessage(String message){
            for (PrintWriter pw:allOut){  //遍历各个客户端
                pw.println("消息广播"+message);//写出消息
            }
        }
        //理解：broadcastMessage（）  功能 ：获取客户端：转发消息
        private void handleChat() {
            new Thread(() -> {
                try {
                    String message;
                    while ((message = br.readLine()) != null) {
                        System.out.println(ip + "服务器端说：" + message);

                        System.out.println("服务器端请输入：");
                      //理解；客户端写出---服务器端读入
                        // 转发消息  客户端写出---服务器端读入的内容
                        broadcastMessage(message);

                    }
                } catch (IOException e) {
                    System.out.println();
                }
            }).start();
            System.out.println("输入文字内容：");
            Scanner input = new Scanner(System.in);
            while (true) {
                String say = input.nextLine();
                pw.println("服务器端说：" + say);
                System.out.println("服务器端输入：");
                if (say.equalsIgnoreCase("exit")){
                    System.exit(0);
                }
            }
        }
        public void run() {
            try {
                //5.当客户端进行读写功能前  加入登录功能
                //交互问题：服务器端向客户端有一个发送文本的功能
                //          客户端向服务器有一个接收功能
                //5.1  向客户端发送一个提示消息：请输入用户名
                pw.println("请输入用户名：");
                //5.2 客户端输入账号【服务器端读取】
                username=br.readLine();
                pw.println("请输入密码：");
                password=br.readLine();
                //5.3 服务器端：判断账号密码 正不正确
                //5.4用户可能有很多个  封装方法  Login  ：判断账号密码
                if (Login(username,password)){
                    pw.println("登录成功  欢迎进入皮皮牌聊天室");
                    //聊天功能    封装方法：读写功能  handleChat
                    handleChat();
                }else {
                    pw.println("登录失败 用户名或者密码错误");
                    socket.close();   //关闭
                }
             //   BufferedReader br = new BufferedReader(new InputStreamReader(socket.getInputStream(), "UTF8"));
              //  pw=new PrintWriter(new BufferedWriter(new OutputStreamWriter(socket.getOutputStream(),"UTF8")),true);

            }catch (IOException e){
                System.out.println("服务器端读写出问题");
            } finally {
                synchronized (allOut){  //目的：移除 刚刚链接的客户端
                    //理解：一个客户端已经转发完毕了  移除当前客户端的写出功能
                    allOut.remove(pw);
                }
                if (pw!=null)
                    pw.close();
                if (br != null)
                    try {
                        br.close();
                    }catch (IOException e){
                        System.out.println();
                    }

                if (socket!=null&&!socket.isClosed()){
                    System.out.println(ip+"一个客户端退出服务器连接");
                    try {
                        socket.close();
                    } catch (IOException e) {
                        System.out.println(ip+"一个客户端退出链接异常");
                    }
                }

                }
            }
        }
    }

